Introduction à la modélisation statistique bayésienne

Ladislas Nalborczyk
LPC, LNC, CNRS, Aix-Marseille Univ.

Planning

Cours n°01 : Introduction à l'inférence bayésienne
Cours n°02 : Modèle Beta-Binomial
Cours n°03 : Introduction à brms, modèle de régression linéaire
Cours n°04 : Modèle de régression linéaire (suite)
Cours n°05 : Markov Chain Monte Carlo
Cours n°06 : Modèle linéaire généralisé
Cours n°07 : Comparaison de modèles
Cours n°08 : Modèles multi-niveaux
Cours n°09 : Modèles multi-niveaux généralisés
Cours n°10 : Data Hackathon

Le problème avec la distribution postérieure

\[ \color{purple}{p(\mu, \sigma | h)} = \frac{\prod_{i} \color{orangered}{\mathrm{Normal}(h_{i} | \mu, \sigma)}\color{steelblue}{\mathrm{Normal}(\mu | 178, 20)\mathrm{Uniform}(\sigma | 0, 50)}} {\color{green}{\int \int \prod_{i} \mathrm{Normal}(h_{i} | \mu, \sigma)\mathrm{Normal}(\mu | 178, 20)\mathrm{Uniform}(\sigma | 0, 50) \mathrm{d} \mu \mathrm{d} \sigma}} \]

Petit problème : \( p(\text{data}) \) s'obtient en calculant la somme (pour des variables discrètes) ou l'intégrale (pour des variables continues) de la densité conjointe \( p(\text{data}, \theta) \) sur toutes les valeurs possibles de \( \theta \). Cela se complique lorsque le modèle comprend plusieurs paramètres.

Le problème avec la distribution postérieure

Rappels cours n°02

Trois méthodes pour résoudre (contourner) ce problème :

  • La distribution a priori est un prior conjugué de la fonction de vraisemblance (e.g., modèle Beta-Binomial). Dans ce cas, il existe une solution analytique (qu'on peut calculer de manière exacte) pour la distribution postérieure.

  • Autrement, pour des modèles simples, on peut utiliser la méthode par grille. On calcule la valeur exacte de la probabilité postérieure en un nombre fini de points dans l'espace des paramètres.

  • Pour les modèles plus complexes, explorer tout l'espace des paramètres n'est pas tractable. On va plutôt échantillonner intelligemment un grand nombre de points dans l'espace des paramètres.

Objectifs du cours

\( \longrightarrow~ \) Présenter le principe de base de l'échantillonnage : Markov Chain Monte Carlo

\( \longrightarrow~ \) Présenter les algorithmes classiques

\( \longrightarrow~ \) Montrer les forces mais aussi les faiblesses de ces méthodes

\( \longrightarrow~ \) Donner des outils de contrôle sur ces méthodes

\( \longrightarrow~ \) Appliquer ces méthodes à un cas simple

Markov Chain Monte Carlo

  • Markov chain Monte Carlo
    \( \longrightarrow~ \) Échantillonnage aléatoire
    \( \longrightarrow~ \) Le résultat est un ensemble de valeurs du paramètre

  • Markov chain Monte Carlo
    \( \longrightarrow~ \) Les valeurs sont générées sous forme de séquences (liaison de dépendance)
    \( \longrightarrow~ \) Indice temporel pour identifier la place dans la chaîne
    \( \longrightarrow~ \) Le résultat est de la forme : \( \theta^1, \theta^2, \theta^3, \dots, \theta^t \)

  • Markov chain Monte Carlo
    \( \longrightarrow~ \) La valeur de paramètre générée ne dépend que de la valeur du paramètre précédent \( \longrightarrow~P(\theta^{t+1}~|~\theta^{t},\theta^{t-1}, \dots,\theta^{1})~=~P(\theta^{t+1}~|~\theta^{t}) \)

Méthodes Monte Carlo

Le terme de méthode de Monte-Carlo désigne une famille d'algorithmes visant à calculer (ou approcher) une valeur numérique en utilisant des procédés aléatoires, c'est-à-dire des techniques probabilistes. Cette méthode a été formalisée en 1947 par Nicholas Metropolis, et publiée pour la première fois en 1949 dans un article co-écrit avec Stanislaw Ulam.

plot of chunk metropolis_picture

Méthodes Monte Carlo : Estimation de \( \pi \)

Soit un point \( M \) de coordonnées \( (x, y) \), où \( 0 < x < 1 \) et \( 0 < y < 1 \). On tire aléatoirement les valeurs de \( x \) et \( y \) entre \( 0 \) et \( 1 \) suivant une loi uniforme. Le point \( M \) appartient au disque de centre \( (0, 0) \) de rayon \( R = 1 \) si et seulement si \( x^{2} + y^{2} \leqslant 1 \). La probabilité que le point \( M \) appartienne au disque est \( \pi / 4 \) puisque le quart de disque est de surface \( \sigma = \frac{\pi R^{2}}{4} = \frac{\pi}{4} \), et le carré qui le contient est de surface \( S = R^{2} = 1 \). Si la loi de probabilité du tirage de point est uniforme, la probabilité de tomber dans le quart de disque vaut \( \frac{\sigma}{S} = \frac{\pi}{4} \). En faisant le rapport du nombre de points dans le disque au nombre de tirages, on obtient une approximation du nombre \( \pi / 4 \) si le nombre de tirages est grand.

plot of chunk pi_gif

Méthodes Monte Carlo : Estimation de \( \pi \)

trials <- 1e5
radius <- 1
x <- runif(n = trials, min = 0, max = radius)
y <- runif(n = trials, min = 0, max = radius)
distance <- sqrt(x^2 + y^2)
inside <- distance < radius
pi_estimate <- 4 * sum(inside) / trials

plot of chunk pi2

Méthodes Monte Carlo

Autre exemple : déterminer la superficie d'un lac ou encore déterminer le maximum d'une fonction (optimisation) via simulated annealing ou recuit simulé (https://en.wikipedia.org/wiki/Simulated_annealing).

plot of chunk simulated_annealing

Méthodes Monte Carlo

Monte Carlo désigne une famille d'algorithmes qui ont pour but d'approcher des valeurs numériques à partir de procédés aléatoires. Pourrait-on s'en servir pour obtenir une approximation de la distribution postérieure ?

On connaît les priors \( p(\theta_{1}) \) et \( p(\theta_{2}) \)
On connaît la fonction de vraisemblance \( p(\text{data} | \theta_{1}, \theta_{2}) \)

Mais on ne sait pas calculer la distribution postérieure… \( p(\theta_{1}, \theta_{2} | \text{data}) = \dfrac{p(\text{data} | \theta_{1}, \theta_{2}) p(\theta_{1}) p(\theta_{1})}{p(\text{data})} \)

Où plutôt on ne sait pas calculer \( p(\text{data}) \)… ! Mais on sait calculer la distribution postérieure à une constante près. Or, comme \( p(\text{data}) \) est une constante, elle ne change pas la forme de la distribution postérieure… ! On va donc explorer l'espace des paramètres et produire des échantillons proportionnellement à leur (densité de) probablité relative.

Influence de la constante de normalisation

Méthodes Monte Carlo : Exemple

Considérons un exemple simple : Soit un paramètre \( \theta \) avec 7 valeurs possibles et la fonction de répartition suivante.

plot of chunk distribution_theta1

Méthodes Monte Carlo : Exemple

Estimation de cette distribution par tirage aléatoire : Cela revient à tirer aléatoirement un grand nombre de points “au hasard” parmi ces 28 cases (comme pour le calcul de \( \pi \)) !

plot of chunk distribution_theta2

Méthodes Monte Carlo : Exemple

plot of chunk unnamed-chunk-3

\( ~\bullet~ \) La distribution des échantillons obtenus converge vers la “vraie” distribution
\( ~\bullet~ \) Mais, nécessite généralement beaucoup d'échantillons (ici pour 7 valeurs de paramètres et 100 tirages)
\( ~\bullet~ \) Aucun contrôle sur la vitesse de convergence…
\( ~\bullet~ \) Et si on abandonnait l'échantillonnage indépendant ?

Algorithme Metropolis

Cet algorithme a été présenté pour la première fois en 1953 par Nicholas Metropolis, Arianna W. Rosenbluth, Marshall Rosenbluth, Augusta H. Teller, et Edward Teller. Le problème des algorithmes Monte-Carlo n'est pas la convergence, mais la vitesse à laquelle la méthode converge. Pour augmenter la vitesse de convergence, il faudrait faciliter l'accès aux valeurs de paramètres les plus représentées.

Principe
\( \bullet~ \) On fait une proposition de déplacement sur la base de la valeur courante du paramètre
\( \bullet~ \) On réalise un tirage aléatoire pour accepter ou rejeter la nouvelle position

Deux idées centrales
\( \bullet~ \) La proposition doit favoriser les valeurs de paramètre les plus probables
\( ~~\longrightarrow~ \) On parcourt plus souvent ces valeurs de paramètres

\( \bullet~ \) La proposition doit se limiter aux valeurs adjacentes au paramètre courant
\( ~~\longrightarrow~ \) On augmente la vitesse de convergence en restant là où se trouve l'information (i.e., en parcourant l'espace des paramètres de manière locale plutôt que globale)

Algorithme Metropolis

1) Sélectionner un point de départ (on peut sélectionner n'importe quelle valeur).

plot of chunk unnamed-chunk-4

Algorithme Metropolis

2) Faire une proposition de déplacement centrée sur la valeur courante de \( \theta \).

plot of chunk unnamed-chunk-5

Algorithme Metropolis

3) Calculer la probabilité d'accepter le déplacement :

\[ \Pr_{\text{move}} = \text{min} \left(\frac{\Pr(\theta_{\text{proposed}})}{\Pr(\theta_{\text{current}})}, 1 \right) \]

plot of chunk unnamed-chunk-6

Algorithme Metropolis

4) Accepter ou rejeter la proposition de déplacement. On compare le ratio de probabilités \( \Pr_{\text{move}} \) à un tirage uniforme. On génère une valeur \( u \) suivant \( u \sim \mathrm{Uniform(0, 1)} \). Ensuite :

\[ \begin{aligned} \text{Si }& u \leq \Pr_{\text{move}} \text{, Alors on accepte le déplacement}\\ \text{Si }& u > \Pr_{\text{move}} \text{, Alors on rejete le déplacement}\\ \end{aligned} \]

plot of chunk unnamed-chunk-7

Algorithme Metropolis

5) La position calculée devient la nouvelle position de départ et on répète l'algorithme.

plot of chunk unnamed-chunk-8

Méthodes Monte Carlo vs. Algorithme Metropolis

plot of chunk metropolis1

plot of chunk metropolis2

Algorithme Metropolis

Application au lancer de pièce (cas continu)

\( ~\bullet~ \) La fonction de vraisemblance est donnée par : \( \color{orangered}{P(z~|~\theta,N)=\theta^z~(1-\theta)^{(N-z)}} \)
\( ~\bullet~ \) Le prior est donné par : \( \color{steelblue}{P(\theta~|~a,b) \propto \theta^{(a-1)}(1-\theta)^{(b-1)}} \)
\( ~\bullet~ \) Le paramètre que l'on cherche à estimer prend ses valeurs dans l'intervalle \( \left[0,1\right] \)

Problème n°1 : Comment définir la proposition de déplacement ?

Le déplacement est modélisé par une distribution normale : \( \Delta \theta \sim \mathrm{Normal}(0, \sigma) \)
\( \longrightarrow~ \) La moyenne \( \mu \) vaut \( 0 \) : le déplacement se fait autour de la valeur courante du paramètre
\( \longrightarrow~ \) La variance reste à déterminer, elle contrôle l'éloignement de la nouvelle valeur

Algorithme Metropolis

Problème n°2 : Quelle probabilité utiliser pour accepter ou refuser le déplacement ? Nous utilisons le produit de la vraisemblance et du prior : \( \color{orangered}{\theta^z~(1-\theta)^{(N-z)}}\color{steelblue}{\theta^{(a-1)}(1-\theta)^{(b-1)}} \)

La probabilité d'accepter le déplacement est donnée par : \( \Pr_{\text{move}} = \text{min} \left(\frac{\Pr(\theta_{\text{current}} + \Delta\theta)}{\Pr(\theta_{\text{current}})}, 1 \right) \)

plot of chunk unnamed-chunk-9

REMARQUE : Le rapport \( \frac{\Pr(\theta_{\text{current}} + \Delta\theta)}{\Pr(\theta_{\text{current}})} \) est le même que l'on utilise la distribution postérieure ou le produit prior par vraisemblance (car la constante de normalisation s'annule) !

Algorithme Metropolis

\( \longrightarrow~ \) Sélectionner un point de départ
\( ~\bullet~ \) Il faut choisir \( ~\theta \in \left[0,1\right] \)
\( ~\bullet~ \) Seule contrainte \( ~\Pr(\theta_{initial}) \ne 0 \)

\( \longrightarrow~ \) Choisir une direction de déplacement
\( ~\bullet~ \) Faire un tirage suivant \( ~\mathrm{Normal}(0, \sigma) \)

\( \longrightarrow~ \) Accepter ou rejeter la proposition de déplacement, suivant la probabilité :

\[ \Pr_{\text{move}} = \text{min} \left(\frac{\Pr(\theta_{\text{current}} + \Delta\theta)}{\Pr(\theta_{\text{current}})}, 1 \right) \]

\( \longrightarrow~ \) La position calculée devient la nouvelle position

Algorithme Metropolis

plot of chunk unnamed-chunk-10

Algorithme Metropolis

Le choix de sigma dans la proposition de déplacement

Deux indices permettent d'évaluer la qualité de l'échantillonnage :

\( \rightarrow \) Le rapport entre le nombre de déplacements proposés et le nombre de déplacements acceptés

\( \rightarrow \) L'effective sample size (i.e., le nombre de déplacements qui ne sont pas corrélés avec les précédents)

Algorithme Metropolis

plot of chunk unnamed-chunk-11

Algorithme Metropolis

Le choix de sigma dans la proposition de déplacement

\( \rightarrow~ \) Toutes les propositions de déplacement (ou presque) sont acceptées

\( \rightarrow~ \) Peu de valeurs effectives

Il va falloir beaucoup d'itérations pour avoir un résultat satisfaisant

Algorithme Metropolis

plot of chunk unnamed-chunk-12

Algorithme Metropolis

Le choix de sigma dans la proposition de déplacement

\( \rightarrow \) Les propositions de déplacement sont rarement acceptées

\( \rightarrow \) Peu de valeurs effectives…

Il va falloir beaucoup d'itérations pour obtenir un résultat satisfaisant

Algorithme Metropolis-Hastings

target <- function(x) {
  # target distribution is Exponential(1)
  # implementation from https://stephens999.github.io/fiveMinuteStats/MH-examples1.html

  if (x < 0) {
    return(0)
    }
  else {
    return(exp(-x) )
    }
}

metropolis_hastings <- function(niter, startval, proposalsd) {

  x <- rep(0, niter) # initialises the chain vector
  x[1] <- startval # defines the starting value

  for(i in 2:niter) {
    currentx <- x[i - 1] # retrieves current value of the parameter
    proposedx <- rnorm(1, mean = currentx, sd = proposalsd) # proposed move
    A <- target(proposedx) / target(currentx) # probability ratio
    if (runif(1) < A) {
      x[i] = proposedx # accept move with probability min(1, A)
      } else {
        x[i] = currentx # otherwise "reject" move, and stay where we are
        }
  }
  return(x)
}

Algorithme Metropolis-Hastings

z1 <- metropolis_hastings(niter = 1000, startval = 3, proposalsd = 1)
z2 <- metropolis_hastings(niter = 1000, startval = 3, proposalsd = 1)

data.frame(z1 = z1, z2 = z2) %>%
  mutate(sample = 1:1000) %>%
  pivot_longer(cols = z1:z2) %>%
  ggplot(aes(x = sample, y = value, colour = name) ) +
  geom_line(show.legend = FALSE) +
  labs(x = "Nombre d'itérations", y = expression(theta) )

plot of chunk metropolis-hastings2

Algorithme Metropolis-Hastings

data.frame(z1 = z1, z2 = z2) %>%
  pivot_longer(cols = z1:z2) %>%
  rownames_to_column() %>%
  mutate(rowname = as.numeric(rowname) ) %>%
  ggplot(aes(x = value) ) +
  geom_histogram() +
  facet_wrap(~name) +
  labs(x = expression(theta), y = "Nombre d'échantillons")

plot of chunk metropolis-hastings3

Algorithme Metropolis-Hastings

Algorithme Gibbs - Modèle bivarié

L'algorithme Gibbs est utilisé pour évaluer la distribution postérieure de modèles composés d'au moins deux variables et diffère de l'algorithme de Metropolis-Hastings en ce qu'il accepte toutes les propositions de déplacements…

Définition du modèle bivarié : On considère une situation dans laquelle on dispose de deux pièces de monnaie. À chaque pièce on associe un paramètre ou biais \( \theta_{1} \) et \( \theta_{2} \) (la probabilité d'obtenir \( \text{Face} \)). On cherche à estimer les biais de ces deux pièces : \( p(\theta_1, \theta_2~|~D) \). Nos variables \( \theta_{1} \) et \( \theta_{2} \) sont indépendantes et identiquement distribuées (iid).

Définition du prior : On considère que chacun des biais est décrit par une distribution Beta : \( \theta_1 \sim \mathrm{Beta}(\theta_1~|~a_1,b_1) \) et \( ~\theta_2 \sim \mathrm{Beta}(\theta_2~|~a_2,b_2) \).

Algorithme Gibbs - Modèle bivarié

On peut donc calculer la distribution a priori de \( P(\theta_1, \theta_2) \) :

\[ \begin{align} p(\theta_1, \theta_2) &= p(\theta_1) p(\theta_2) \\ &= \theta_1^{(a_1-1)}~(1-\theta_1)^{(b_1-1)}\theta_2^{(a_2-1)}~(1-\theta_2)^{(b_2-1)} \end{align} \]

plot of chunk bivariate_prior

Algorithme Gibbs - Modèle bivarié

Description des données : On dispose d'un ensemble de lancers pour les deux pièces. On appelle \( ~D_1 \) les données issues de la première pièce, composé de \( ~z_1 \) Face sur \( ~N_1 \) lancers. On appelle \( ~D_2 \) les données issues de la seconde pièce, composé de \( ~z_2 \) Face sur \( ~N_2 \) lancers.

On note les données : \( D = \{ z_1, N_1, z_2, N_2 \} \)

Définition de la fonction de vraisemblance

\[ \begin{align} P(D~|~\theta_{1}, \theta_{2}) &= \prod_{y_{1i} \in D_{1}} P(y_{1i}~|~\theta_{1},\theta_{2})~ \prod_{y_{2i} \in D_{2}} P(y_{2i}~|~\theta_{1},\theta_{2}) &&\mbox{Règle d'indépendance}\\ &= \theta_1^{z_{1}}~(1 - \theta_{1})^{(N{1} - z_{1})}\theta_{2}^{z_2}~(1 - \theta_{2})^{(N_{2} - z_{2})}~ \end{align} \]

Algorithme Gibbs - Modèle bivarié

Résolution analytique

\[ \begin{align} \color{purple}{p(\theta_1,\theta_2~|~D)} &= \color{orangered}{p(D~|~\theta_1,\theta_2)}~\color{steelBlue}{p(\theta_1,\theta_2)}/\color{green}{p(D)} \\ &= \color{orangered}{\theta_1^{z_1}~(1-\theta_1)^{(N1-z_1)}\theta_2^{z_2}~(1-\theta_2)^{(N2-z_2)}}~\color{steelBlue}{p(\theta_1,\theta_2)}/\color{green}{p(D)} \\ &= \frac{\theta_1^{(\color{orangered}{z_1}+\color{steelBlue}{a_1-1})}(1-\theta_1)^{(\color{orangered}{N_1-z_1}+\color{steelBlue}{b_1-1})}\theta_2^{(\color{orangered}{z_2}+\color{steelBlue}{a_2-1})}(1-\theta_2)^{(\color{orangered}{N_2-z_2}+\color{steelBlue}{b_2-1})}~}{\color{green}{p(D)}~\color{steelBlue}{B(a_1,b_1)~B(a_2,b_2)}} \end{align} \]

Donc finalement :

\[ \begin{align} \color{purple}{p(\theta_1,\theta_2~|~D)} =& \ \color{purple}{\mathrm{Beta}(\theta_1~|~z_1+a_1,N_1-z_1+b_1)} \color{purple}{\mathrm{Beta}(\theta_2~|~z_2+a_2,N_2-z_2+b_2)} \end{align} \]

On voit que lorsque le prior est le produit de deux distributions Beta indépendantes, alors la distribution postérieure est également le produit de deux distributions Beta indépendantes.

Algorithme Gibbs - Modèle bivarié

plot of chunk bivariate

Modèle à deux variables - Algorithme Metropolis

  • La proposition de saut repose sur une une loi normale bivariée

\( \longrightarrow \) La position suivante peut-être n'importe où sur la grille

plot of chunk metropolis-2d

Algorithme Gibbs - Description

On voit que l'efficacité de l'algorithme Metropolis repose sur la proposition de déplacement. Cette fonction peut-être difficile à calibrer :

\( \longrightarrow~ \) On voudrait que le SD soit petit dans les zones à forte densité
\( \longrightarrow~ \) Au contraire, on voudrait que le SD soit grand dans les zones à faible densité

L'algorithme Gibbs résout ce problème. L'algorithme de Gibbs parcourt l'espace des paramètres comme l'algorithme Metropolis, mais plutôt que de modifier tous les paramètres en même temps, il ne modifie qu'un paramètre après l'autre. La proposition de déplacement est donc basée sur la fonction de densité conditionnelle. Le déplacement est toujours accepté, mais l'algorithme requiert que l'on puisse calculer la densité postérieure conditionnelle…

Algorithme Gibbs - Description

\( \longrightarrow~ \) Sélectionner un point de départ \( (\theta_{1init},\theta_{2init}) \)
On peut sélectionner n'importe quelle valeur

\( \longrightarrow~ \) Sélectionner le premier paramètre par exemple \( \theta_1 \)

\( \longrightarrow~ \) Générer une valeur aléatoire de \( \theta_1 \)
Faire un tirage directement suivant la probabilité \( p(\theta_{1}~|~\theta_{2}, D) \)

\( \longrightarrow~ \) Générer une valeur aléatoire de \( \theta_2 \)
Faire un tirage suivant la probabilité \( p(\theta_{2}~|~\theta_{1}, D) \)
Avec pour valeur de \( \theta_{1} \) la valeur calculée précédemment

\( \longrightarrow~ \) On boucle sur les deux dernières étapes pour parcourir

Algorithme Gibbs - Description

plot of chunk gibbs1

Algorithme Gibbs - Description

plot of chunk gibbs2

On voit que l'effective sample size est beaucoup plus grand que celui obtenu avec l'algorithme Metropolis…

Algorithme Gibbs - Implémentation

On suppose que \( Y \sim \mathrm{Normal}(\mu, \frac{1}{\tau}) \). On souhaite approximer la distribution postérieure de \( \mu \) et \( \tau \) en utilisant l'algorithme Gibbs. On note que \( \mu \) représente la moyenne dans la population, \( \tau \) représente la précision (réciproque de la variance) dans la population, \( n \) représente la taille d'échantillon, \( \bar{y} \) représente la moyenne dans l'échantillon et \( s^{2} \) la variance de l'échantillon. L'algorithme Gibbs peut être résumé de la manière suivante (from Casella & Georges, 1992) :

  • Échantillonner \( \mu^{(i)} \) à partir de \( \ p(\mu \ | \ \tau^{(i - 1)}, \text{données}) \)
  • Échantillonner \( \tau^{(i)} \) à partir de \( \ p(\tau \ | \ \mu^{(i)}, \text{données}) \)

On définit les priors suivants : \( p(\mu, \tau) = p(\mu) \times p(\tau) \) où \( p(\mu) \propto 1 \) et \( p(\tau) \propto \tau^{-1} \).

Dans cette configuration la distribution postérieure conditionnelle pour la moyenne, sachant la précision, est donnée par \( (\mu \mid \tau, \text { data }) \sim \mathrm{Normal}\left(\bar{y}, \frac{1}{n \tau}\right) \). La distribution postérieure conditionnelle pour la précision, sachant la moyenne, est donnée par \( (\tau \mid \mu, \text { data }) \sim \operatorname{Gamma}\left(\frac{n}{2}, \frac{2}{(n-1) s^{2}+n(\mu-\bar{y})^{2}}\right) \).

Algorithme Gibbs - Implémentation

# code from https://stats.stackexchange.com/questions/266665/gibbs-sampler-examples-in-r
n <- 30 # sample size
ybar <- 15 # sample mean
s2 <- 3 # sample variance

mu <- rep(NA, 11000) # initialises mu vector
tau <- rep(NA, 11000) # initialises tau vector

burn <- 1000 # burnin period
tau[1] <- 1 # initialisation value for tau

# samples from the joint posterior (mu, tau | data)
for(i in 2:11000) {

  mu[i]  <- rnorm(n = 1, mean = ybar, sd = sqrt(1 / (n * tau[i - 1]) ) )    
  tau[i] <- rgamma(n = 1, shape = n / 2, scale = 2 / ((n - 1) * s2 + n * (mu[i] - ybar)^2) )

}

mu  <- mu[-(1:burn)] # removes burnin
tau <- tau[-(1:burn)] # removes burnin

Algorithme Gibbs - Implémentation

data.frame(mu = mu, tau = tau) %>%
  pivot_longer(cols = mu:tau) %>%
  ggplot(aes(x = value) ) +
  geom_histogram() +
  facet_wrap(~name, scales = "free") +
  labs(x = "Valeur du paramètre", y = "Nombre d'échantillons")

plot of chunk gibbs2_code2

Algorithme Gibbs

Algorithme Gibbs - Conclusions

  1. L'algorithme Gibbs est particulièrement utile lorsque la distribution jointe \( p(\{\theta_i\} | \text{data}) \) n'est pas calculable ou n'est pas facile à échantillonner, mais que les densités conditionnelles \( p(\theta_i | \{\theta_{j \ne i}\}, \text{data}) \) le sont.

  2. L'algorithme Gibbs peut être vu comme un cas particulier de l'algorithme Metropolis pour lequel la proposition de saut dépend de la position dans l'espace des paramètres et du paramètre sélectionné.

  3. En général, l'algorithme Gibbs améliore les performances de l'algorithme Metropolis.

Algorithme Hamiltonian Monte Carlo

On a vu que l'algoritme Gibbs améliore les performances de convergence de l'algorithme Metropolis. Mais il nécessite de pouvoir calculer les densités conditionnelles \( p(\theta_i | \theta_{j \ne i}, \text{data}) \). L'algorithme est performant lorsqu'il y a indépendance des paramètres \( p(\theta_i | \theta_{j \ne i}, \text{data}) = p(\theta_i | \text{data}) \)… mais ce n'est pas le cas en général !

plot of chunk gibbs_error

Algorithme Hamiltonian Monte Carlo

Les algorithmes Metropolis-Hastings et Gibbs ont de mauvaises performances lorsque les paramètres du modèle sont fortement corrêlés. L'algorithme Hamiltonian Monte Carlo repose sur l'algorithme Metropolis mais résout ces problème en utilisant la géométrie de l'espace postérieur. On utilise l'opérateur hamiltonien (hamiltonians) qui représente l'énergie totale d'un système. Cette énergie se décompose en l'énergie potentielle (qui dépend de la position dans l'espace des paramètres \( \theta \)) et son énergie cinétique, qui dépend de son momentum (\( m \)) :

\[ H(\theta, m) = \underbrace{U(\theta)}_{\text{énergie potentielle}} + \underbrace{KE(m)}_{\text{énergie cinétique}} \]

L'énergie potentielle est donnée par le négatif du log de la densité postérieure (non-normalisée) ;

\[ U(\theta) = -\log[p(data | \theta) \times p(\theta)] \]

Quand la densité postérieure augmente, l'énergie potentielle diminue (i.e., devient plus négative).

Algorithme Hamiltonian Monte Carlo

Si on “suit” simplement la gravité de notre position dans l'espace postérieur (imaginons que nous suivions la trajectoire d'une luge ou d'une bille dans cet espace), on arrivera à un minimum de cet espace, mais on veut explorer tout l'espace proportionnellement à la densité postérieure, pas seulement les endroits de forte densité…

L'algorithme Metropolis repose sur les principes suivants :

\( \longrightarrow~ \) La proposition de déplacement est toujours une gaussienne multivariée centrée sur la position courante.

\( \longrightarrow~ \) La proposition de déplacement est une gaussienne qui garde toujours la même forme (variance constante).

L'algorithme HMC repose sur une autre idée : Adapter la proposition de déplacement en fonction de la position courante du paramètre et de la géométrie du postérieur aux alentours de cette position.

Algorithme Hamiltonian Monte Carlo

plot of chunk hmc1

Algorithme Hamiltonian Monte Carlo

\( \longrightarrow~ \) Sélectionner un point de départ \( \theta_{0} \) : On peut sélectionner n'importe quelle valeur de \( \theta \) dans l'espace des paramètres

\( \longrightarrow~ \) Générer une nouvelle position à partir de la fonction de potentiel
\( ~\bullet~ \) L'algorithme génère une proposition par analogie au lancer d'une bille sur la fonction de potentiel
\( ~\bullet~ \) Le pas de déplacement, ainsi que le nombre de déplacement, sont fixés à l'avance
\( ~\bullet~ \) La force avec laquelle on lance la bille est déterminée par une loi normale multivariée : \( m \sim \mathrm{MVNormal}(\mu, \Sigma) \)
\( ~\bullet~ \) Estimation de la trajectoire discrétisée via la méthode leapfrog

\( \longrightarrow~ \) Accepter ou rejeter la proposition de déplacement suivant la probabilité (où \( \phi \) est le moment associé à la bille) :

\[ \Pr_{déplacement} = \min \left(\frac{\Pr(\theta_{proposé}~|~D)~\Pr(\phi_{final})}{\Pr(\theta_{courant}~|~D)~\Pr(\phi_{initial})}, 1 \right) \]

\( \longrightarrow~ \) On enregistre la nouvelle position et on recommence…

Influence de la durée de déplacement...

plot of chunk hmc_erreur

Influence de la variabilité du moment initial...

plot of chunk hmc_erreur2

Algorithme Hamiltonian Monte Carlo

Algorithme Hamiltonian Monte Carlo

  1. L'algorithme HMC est particulièrement utile lorsque le modèle contient un grand nombre de paramètres et quand les paramètres sont corrélés entre eux.

  2. Il nécessite un paramétrage assez fin. Des logiciels proposent des méthodes clés en main très efficaces.

  3. Le package brms utilise une variation de cet algorithme (l'algorithme NUTS) qui permet d'éviter les trajectoires cycliques.

Évaluation des MCMC

Ces méthodes peuvent ne pas converger vers la “vraie” distribution postérieure, en raison du temps de calcul limité, du paramétrage de certains hyper-paramètres (e.g., variance de la distribution normale de la proposition pour Gibbs, ou variance du moment initial pour HMC).

Ces méthodes produisent des chaînes de valeurs de paramètres (échantillons). L'utilisation de tel ou tel algorithme MCMC pour échantillonner la distribution postérieure repose sur trois objectifs :

  1. Les valeurs de la chaîne doivent être représentatives de la distribution postérieure
    \( ~\bullet~ \) Ces valeurs ne doivent pas dépendre du point de départ
    \( ~\bullet~ \) Ces valeurs ne doivent pas être cantonnées à une région particulière de l'espace des paramètres

  2. La chaîne doit être suffisamment longue pour assurer la précision et la stabilité du résultat
    \( ~\bullet~ \) La tendance centrale et le HDI calculés à partir de la chaîne ne doivent pas changer si on relance la procédure

  3. La chaîne doit être générée de manière efficace (avec le moins d'itérations possible)

Évaluation des MCMC - Représentativité

  • Vérification visuelle des trajectoires : Les chaînes doivent occuper le même espace, la convergence ne dépend pas du point de départ, aucune chaîne ne doit avoir de trajectoire particulière (e.g., cyclique).

  • Vérification visuelle des densités : Les densités doivent se superposer.

plot of chunk repres1

Évaluation des MCMC - Représentativité

Cette affichage ne montre que les 500 premières itérations. Les trajectoires ne se superposent pas au début (zone orange). La densité est également affectée. En pratique on supprime ces premières itérations (burn-in ou warm-up).

plot of chunk repres2

Évaluation des MCMC - Représentativité

Vérification numérique des chaînes : Le shrink factor (aussi connu comme \( \hat{R} \) ou Rhat) est le rapport entre la variance inter-chaînes et intra-chaîne. Cette valeur devrait idéalement tendre vers 1 (on la considère comme acceptable jusqu'à 1.01).

plot of chunk repres3

Évaluation des MCMC - Stabilité et précicion

Plus la chaîne est longue et plus le résultat sera précis et stable. Si la chaîne s'attarde sur chaque position, et que le nombre d'itérations reste le même, alors on perd en précision. Il lui faudra plus d'itérations pour arriver au même niveau de précision. L'autocorrélation est la corrélation de la chaîne avec elle-même mais décalé de \( k \) itérations (lag).

plot of chunk autocorrelation

Évaluation des MCMC - Stabilité et précicion

La fonction d'autocorrélation est représentée pour chaque chaîne (en haut à droite). Un autre résultat rend compte de la précision de l'échantillon : l'effective sample size, \( ESS = \frac{N}{1 + 2 \sum_k ACF(k)} \). Il représente la taille d'un échantillon non-autocorrélé extrait de la somme de toutes les chaînes. Pour une précision raisonnable du HDI, il est recommandé d'avoir un ESS supérieur à 1000.

plot of chunk repres4

Évaluation des MCMC - Stabilité et précicion

L'erreur standard d'un ensemble d'échantillons est donné : \( SE = SD / \sqrt{N} \). Plus \( N \) augmente plus l'erreur standard diminue. On peut généraliser cette idée aux chaînes de Markov : \( MCSE = SD / \sqrt{ESS} \). Pour une précision raisonnable de la tendance centrale, il faut que cette valeur soit faible.

plot of chunk repres5

Évaluation des MCMC - Implémentation via brms

library(rethinking)
library(tidyverse)
library(brms)

data(Howell1)
d <- Howell1
d2 <- d %>% filter(age >= 18)

priors <- c(
  set_prior("normal(150, 20)", class = "Intercept"),
  set_prior("normal(0, 10)", class = "b"),
  set_prior("exponential(0.01)", class = "sigma")
  )

mod1 <- brm(
  height ~ 1 + weight,
  prior = priors,
  family = gaussian(),
  data = d2, 
  chains = 4, # nombre de MCMCs
  iter = 2000, # nombre total d'itérations (par chaîne)
  warmup = 1000, # nombre d'itérations pour le warm-up
  thin = 1 # thinning (1 = no thinning)
  )

Évaluation des MCMC - Implémentation via brms

# combo can be hist, dens, dens_overlay, trace, trace_highlight...
# cf. https://mc-stan.org/bayesplot/reference/MCMC-overview.html

plot(
  x = mod1, combo = c("dens_overlay", "trace"),
  theme = theme_bw(base_size = 16, base_family = "Open Sans")
  )

plot of chunk diagnostics2

Évaluation des MCMC - Implémentation via brms

library(bayesplot)
post <- posterior_samples(mod1, add_chain = TRUE)
post %>% mcmc_acf(pars = vars(b_Intercept:sigma), lags = 10)

plot of chunk diagnostics3

Évaluation des MCMC - Implémentation via brms

summary(mod1)
 Family: gaussian 
  Links: mu = identity; sigma = identity 
Formula: height ~ 1 + weight 
   Data: d2 (Number of observations: 352) 
  Draws: 4 chains, each with iter = 2000; warmup = 1000; thin = 1;
         total post-warmup draws = 4000

Population-Level Effects: 
          Estimate Est.Error l-95% CI u-95% CI Rhat Bulk_ESS Tail_ESS
Intercept   113.85      1.93   110.11   117.60 1.00     4112     2918
weight        0.91      0.04     0.82     0.99 1.00     3962     3012

Family Specific Parameters: 
      Estimate Est.Error l-95% CI u-95% CI Rhat Bulk_ESS Tail_ESS
sigma     5.11      0.19     4.74     5.51 1.00     4124     3036

Draws were sampled using sampling(NUTS). For each parameter, Bulk_ESS
and Tail_ESS are effective sample size measures, and Rhat is the potential
scale reduction factor on split chains (at convergence, Rhat = 1).

Évaluation des MCMC - Implémentation via brms

Bulk-ESS fait référence à l'ESS calculé sur la distribution des échantillons normalisée par leur rang, et plus particulièrement autour de la position centrale de cette distribution (e.g., moyenne ou médiane). On recommande que le Bulk-ESS soit au moins 100 fois plus élevé que le nombre de chaînes (i.e., pour 4 chaînes, le Bulk-ESS devrait être d'au moins 400).

Tail-ESS donne le minimum de l'ESS calculé pour les quantiles à 5% et 95% (i.e., pour les queues de la distribution des échantillons normalisés par leur rang). Cette valeur doit être élevée si nous accordons de l'importance à l'estimation des valeurs extrêmees (par exemple pour calculer un intervalle de crédibilité).

Quand tout va mal, voir ces recommendations de l'équipe de Stan concernant les choix de prior, ou ce guide concernant les messages d'erreur fréquents. Voir aussi l'article récent ou cet article de blog introduisant ces nouveaux indices.

Évaluation des MCMC - Implémentation via brms

post %>% # rank plots
  mcmc_rank_overlay(pars = vars(b_Intercept:sigma) ) +
  labs(x = "Rang", y = "Fréquence") +
  coord_cartesian(ylim = c(25, NA) )

plot of chunk diagnostics5

Résumé du cours

Nous avons introduit et discuté l'utilisation d'échantillonneurs (samplers) pour obtenir des échantillons issus de la distribution postérieure (non-normalisée). Ces échantillons peuvent ensuite être utilisés pour calculer différentes statistiques sur la distribution postérieure (e.g., moyenne, médiane, HDI).

L'algorithme Metropolis-Hastings peut être utilisé pour n'importe quel problème pour lequel une vraisemblance peut être calculée. Cependant, bien que cet algorithme soit simple à coder, sa convergence peut être très lente… L'algorithme Gibbs accélère la convergence mais nécessite qu'on l'on puisse calculer la distribution conditionnelle des paramètres. De plus, cet algorithme ne fonctionne pas bien lorsqu'il existe de fortes corrélations entre les différents paramètres…

L'algorithme HMC évite ces problèmes en prenant en considération la géométrie de l'espace postérieur lors de son exploration (i.e., lorsque l'algorithme décide où il doit aller ensuite). Cet algorithme converge beaucoup plus rapidement que les deux précédents et moins d'échantillons seront nécessaires pour approcher la distribution postérieure.

Le résultat d'une inférence bayésienne est donc, en pratique, un ensemble d'échantillons obtenus en utilisant des MCMCs. La fiabilité des ces estimations doit être évaluée en vérifiant (visuellement et numériquement) que les MCMCs ont bien convergé vers une solution optimale.

Travaux pratiques

On s'intéresse à la performance économique des capitales rgdppc_2000 en fonction de deux paramètres : la rudesse du paysage (plus ou moins vallonné) rugged et son appartenance au continent africain cont_africa.

library(rethinking)
library(tidyverse)

data(rugged)
d <- rugged
d$log_gdp <- log(d$rgdppc_2000)
df1 <- d[complete.cases(d$rgdppc_2000), ]
str(df1[, 1:5])
'data.frame':   170 obs. of  5 variables:
 $ isocode    : Factor w/ 234 levels "ABW","AFG","AGO",..: 3 5 8 9 10 12 13 14 15 16 ...
 $ isonum     : int  24 8 784 32 51 28 36 40 31 108 ...
 $ country    : Factor w/ 234 levels "Afghanistan",..: 6 2 219 9 10 8 12 13 14 34 ...
 $ rugged     : num  0.858 3.427 0.769 0.775 2.688 ...
 $ rugged_popw: num  0.714 1.597 0.316 0.22 0.934 ...

Travaux pratiques

Écrire le modèle qui prédit log_gdp en fonction de la rudesse du terrain, du continent, et de l'interaction de ces deux variables avec brms::brm(), en spécifiant vos propres priors. Examinez ensuite les estimations de ce modèle (interprétation, diagnostiques des MCMCs).

\[ \begin{aligned} \log(\text{gdp}_{i}) &\sim \mathrm{Normal}(\mu_{i}, \sigma_{i}) \\ \mu_{i} &= \alpha + \beta_{1} \cdot \text{rugged}_{i} + \beta_{2} \cdot \text{continent}_{i} + \beta_{3} \cdot (\text{rugged}_{i} \cdot \text{continent}_{i}) \\ \alpha &\sim \cdots \\ \beta_{1}, \beta_{2}, \beta_{3} &\sim \cdots \\ \end{aligned} \]

Proposition de réponse

priors2 <- c(
  set_prior("normal(0, 100)", class = "Intercept"),
  set_prior("normal(0, 10)", class = "b"),
  set_prior("exponential(0.01)", class = "sigma")
  )

mod2 <- brm(
  log_gdp ~ 1 + rugged * cont_africa,
  prior = priors2,
  family = gaussian(),
  data = df1,
  chains = 4, # nombre de MCMCs
  iter = 2000, # nombre total d'itérations (par chaîne)
  warmup = 1000 # nombre d'itérations pour le warm-up
  )

Proposition de réponse

summary(mod2)
 Family: gaussian 
  Links: mu = identity; sigma = identity 
Formula: log_gdp ~ 1 + rugged * cont_africa 
   Data: df1 (Number of observations: 170) 
  Draws: 4 chains, each with iter = 2000; warmup = 1000; thin = 1;
         total post-warmup draws = 4000

Population-Level Effects: 
                   Estimate Est.Error l-95% CI u-95% CI Rhat Bulk_ESS Tail_ESS
Intercept              9.22      0.14     8.94     9.50 1.00     2798     2875
rugged                -0.20      0.08    -0.36    -0.05 1.00     2594     2744
cont_africa           -1.95      0.23    -2.42    -1.49 1.00     2675     2674
rugged:cont_africa     0.39      0.14     0.13     0.66 1.00     2505     2568

Family Specific Parameters: 
      Estimate Est.Error l-95% CI u-95% CI Rhat Bulk_ESS Tail_ESS
sigma     0.95      0.05     0.86     1.06 1.00     4043     3002

Draws were sampled using sampling(NUTS). For each parameter, Bulk_ESS
and Tail_ESS are effective sample size measures, and Rhat is the potential
scale reduction factor on split chains (at convergence, Rhat = 1).

Proposition de réponse

plot(
  x = mod2, combo = c("dens_overlay", "trace"), pars = "^b_",
  theme = theme_bw(base_size = 16, base_family = "Open Sans")
  )

plot of chunk mod2-diagnostics

Proposition de réponse

pairs(mod2)

plot of chunk mod2-pairs